#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@create Time:2018-07-30

@author:Brook
"""
import os
import sys


def test_path(path):
    print("------------")
    print("扫描文件路径: %s" % path)
    if not os.path.exists(path):
        print("该路径不存在!!")
        sys.exit(1)
    else:
        print("检测成功!")
    print("------------")



if __name__ == "__main__":

    test_dir = "/home/data"
    cache = "cache"  # 存放程序的中间结果


    print("清空缓存结果...")
    os.system("rm -rf %s/*" % cache)

    """ 增减持"""
    html_dir_zjc = os.path.join(test_dir, 'zengjianchi/html')
    pubdate_path_zjc = os.path.join(test_dir, 'zengjianchi/zengjianchi_public.xlsx')
    save_path_zjc_table = os.path.join(cache, "zengjianchi_table.txt")
    save_path_zjc_text = os.path.join(cache, "zengjianchi_text.txt")
    # 检测文件是否存在
    test_path(html_dir_zjc)
    test_path(pubdate_path_zjc)

    # 抽取增减持的表格部分数据
    print("抽取增减持表格部分...")
    os.system("python3 extract_zengjianchi_table.py %s %s %s" % (html_dir_zjc, pubdate_path_zjc, save_path_zjc_table))

    # 抽取增减持文本部分的数据
    print("抽取增减持文本部分...")
    os.system("python3 extract_zengjianchi_text.py %s %s" % (html_dir_zjc, save_path_zjc_text))

    # 合并增减持的表格数据与文本数据



